#!/bin/bash

# 创建目录
mkdir -p mistral7b/Mistral-7B-v0.1
cd mistral7b/Mistral-7B-v0.1

# 尝试不同的镜像源
MIRRORS=(
    "https://huggingface.co"
    "https://cdn-lfs.huggingface.co"
    "https://hf-mirror.com"
)

# 需要下载的文件
FILES=(
    "config.json"
    "tokenizer.model"
    "tokenizer.json"
    "tokenizer_config.json"
    "special_tokens_map.json"
    "generation_config.json"
)

# 尝试不同的下载方法
download_file() {
    local file=$1
    local success=0

    # 方法1: 直接使用 wget
    echo "尝试下载 $file..."
    wget -q --no-check-certificate "https://huggingface.co/mistralai/Mistral-7B-v0.1/raw/main/$file" -O "$file" && success=1

    # 方法2: 如果失败，尝试 curl
    if [ $success -eq 0 ]; then
        curl -sL "https://huggingface.co/mistralai/Mistral-7B-v0.1/raw/main/$file" -o "$file" && success=1
    fi

    if [ $success -eq 1 ]; then
        echo "✓ 成功下载: $file"
    else
        echo "✗ 下载失败: $file"
    fi
}

# 下载所有文件
for file in "${FILES[@]}"; do
    download_file "$file"
done

echo -e "\n下载完成，文件列表："
ls -lah